# -*- coding: utf-8 -*-
# @Time    : 2020/6/27 下午11:56
# @Author  : lilong

from rasa.importers.rasa import RasaFileImporter

importer_config = \
    {
        'language': 'en',

        'pipeline': [{'name': 'WhitespaceTokenizer'},
                     {'name': 'RegexFeaturizer'},
                     {'name': 'LexicalSyntacticFeaturizer'},
                     {'name': 'CountVectorsFeaturizer'},
                     {'name': 'CountVectorsFeaturizer', 'analyzer': 'char_wb', 'min_ngram': 1, 'max_ngram': 4},
                     {'name': 'DIETClassifier', 'epochs': 100},
                     {'name': 'EntitySynonymMapper'},
                     {'name': 'ResponseSelector', 'epochs': 100}],

        'policies': [{'name': 'MemoizationPolicy'},
                     {'name': 'TEDPolicy', 'max_history': 5, 'epochs': 100},
                     {'name': 'MappingPolicy'}]}

module_path = importer_config.pop("name", None)
print("module_path:", module_path)

print(RasaFileImporter.__name__)

# if module_path == RasaFileImporter.__name__:
#     importer_class = RasaFileImporter
